package tests.base;

import base.Atendimento;
import base.Medico;
import base.Paciente;
import base.Prontuario;

import org.junit.*;

public class AtendimentoTest{

	private Atendimento a;
	
	@Before
	public void init() {
		a = new Atendimento(1);
	}
	
	@After
	public void deinit() {
		a = null;
	}
	
	@Test
	public void Medico() {
		a.setMedico(new Medico(1));
		Assert.assertNotNull(a.getMedico());
		a.getMedico().setCpf("");
		Medico m = new Medico(1);
		m.setCpf("");
		Assert.assertEquals(a.getMedico(), m);
	}

	@Test
	public void Paciente() {
		a.setPaciente(new Paciente(1));
		Assert.assertNotNull(a.getPaciente());
		a.getPaciente().setCpf("");
		Paciente p = new Paciente(1);
		p.setCpf("");
		Assert.assertEquals(a.getPaciente(), p);
	}

	@Test
	public void Prontuario() {
		Assert.assertNotNull(a.getProntuarios());
		
		a.adicionarProntuario(new Prontuario(3));
		
		Assert.assertEquals(a.getProntuarios().size(), 1);
		Assert.assertNotNull(a.getProntuarios().get(0));

		a.removerProntuario(new Prontuario(3));
		Assert.assertEquals(a.getProntuarios().size(), 0);
	}
}